import pytest
def add(a,b):
    return a+b
def sub(a,b):
    return a-b
def test_add():
    assert add(1,1) == 2
def test_sub():
    assert sub(2,1) == 1
pytest.main()
